用GO搭建一个Web服务器

有人曾和我说过,一门语言应该能够自己实现一个HTTP服务。PHP做不到,但GO却轻而易举。
只需要几行代码,GO就可以实现一个简单的HTTP服务。

package main

import (
    "fmt"
    "net/http"
)

func IndexHandler(w http.ResponseWriter, r *http.Request) {
    fmt.Fprintln(w, "hello world")
}

func main() {
    http.HandleFunc("/", IndexHandler)
    http.ListenAndServe(":8001", nil)
}

这样,即可监听8001端口,并且当有请求访问的时候完成响应。(这几乎就完成了nginx的主要功能,不过没有做负载平衡)
观察这段代码,我们基本能够从命名大概了解每个函数的作用。

http.HandleFunc("/", IndexHandler)

这相当于绑定url的函数,当请求的url是"/"时,那么将这个请求交给IndexHandler来处理。

http.ListenAndServe(":8001", nil)

顾名思义,监听8001端口并且启动服务。

func IndexHandler(w http.ResponseWriter, r *http.Request) {
    fmt.Fprintln(w, "hello world")
}

处理函数,当有请求的时候输出hello world。
通过以上分析,GO搭建一个HTTP服务只需用到一个net/http包即可。同时结合上篇文章,要完成http服务需要以下几点

  • Request 用户请求的信息,用来解析用户的请求信息,包括post、get、cookie、url等信息
  • Response 服务端反馈给客户端的信息
  • Conn 用户每次请求的连接
  • Handler 处理请求和生产返回信息的处理逻辑

前3点是一个HTTP服务必须要的结构,第4点是每一个HTTP服务的核心所在。
我们只需要了解3个问题,就知道GO是如何将Web服务运作起来。

  • 如何监听接口
  • 如何接受客户端请求
  • 如何分配handle

如何监听接口

GO是通过一个函数 ListenAndServe做到的。
首先初始化一个server对象,然后调用net.Listen("tcp", addr)底层建立TCP连接,监听我们设置的端口。

如何接收客户端请求

这块的源码

func (srv *Server) Serve(l net.Listener) error {
    defer l.Close()
    var tempDelay time.Duration // how long to sleep on accept failure
    for {
        rw, e := l.Accept()
        if e != nil {
            if ne, ok := e.(net.Error); ok && ne.Temporary() {
                if tempDelay == 0 {
                    tempDelay = 5 * time.Millisecond
                } else {
                    tempDelay *= 2
                }
                if max := 1 * time.Second; tempDelay > max {
                    tempDelay = max
                }
                log.Printf("http: Accept error: %v; retrying in %v", e, tempDelay)
                time.Sleep(tempDelay)
                continue
            }
            return e
        }
        tempDelay = 0
        c, err := srv.newConn(rw)
        if err != nil {
            continue
        }
        go c.serve()
    }
}

这段代码首先起了一个for()。接收请求Accept(),若有请求,则建立一个Conn连接,并用协程启动服务。这时再次进入下一个循环,若有请求则新建立一个连接并启动一个协程服务。若没有请求,则休息一段时间后再次循环。正是利用了go协程特性,用户的每一个请求都有一个新的goroutine去服务,互不影响,达到了天然支持高并发特性。

如何分配handle

在建立conn之后,conn会首先解析request,c.readRequest(),然后获得相应的handle:handler := c.server.Handler,也就是我们刚才在调用函数ListenAndServe时候的第二个参数,我们前面例子传递的是nil,也就是为空,那么默认获取handler = DefaultServeMux,那么这个变量用来做什么的呢?对,这个变量就是一个路由器,它用来匹配url跳转到其相应的handle函数,那么这个我们有设置过吗?有,我们调用的代码里面第一句不是调用了http.HandleFunc("/", IndexHandler)嘛。这个作用就是注册了请求/的路由规则,当请求uri为"/",路由就会转到函数IndexHandlerDefaultServeMux会调用ServeHTTP方法,这个方法内部其实就是调用IndexHandler本身,最后通过写入response的信息反馈到客户端。


HammerMax
128 声望17 粉丝